from problem_generator import ProblemGenerator
from multi_robot_problem import MultiRobotState

height, width = 10,10
minRobs, maxRobs = 3,5
difficulty = ProblemGenerator.HARD
minPiles, maxPiles, minObs, maxObs = 3, 5, 3, 7 
minWallLength, maxWallLength = 6,8

robots = ((1,1),)
obstacles = frozenset([(2,2),(2,3),(2,4),(3,2),(3,4),(4,2),(4,3),(4,4)])
piles = frozenset([(3,3)])


pGen = ProblemGenerator(height, width, minRobs, maxRobs, difficulty,
    minPiles, maxPiles, minObs, maxObs, minWallLength, maxWallLength)
problem = pGen.trollProblem()
#problem = MultiRobotState(10,10, robots, piles, obstacles)
print problem